package com.lagou.edu.factory;

import com.lagou.edu.anno.Autowired;
import com.lagou.edu.anno.Service;
import com.lagou.edu.anno.Transactional;
import com.lagou.edu.utils.ClassUtils;
import com.mysql.jdbc.StringUtils;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;

import java.io.InputStream;
import java.io.InputStreamReader;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @author 应癫
 *
 * 工厂类，生产对象（使用反射技术）
 */
@Service
public class BeanFactory {


    /**
     * 任务一：读取解析xml，通过反射技术实例化对象并且存储待用（map集合）
     * 任务二：对外提供获取实例对象的接口（根据id获取）
     */

    private static Map<String,Object> map = new HashMap<>();  // 存储对象

//    static {
//        // 任务一：读取解析xml，通过反射技术实例化对象并且存储待用（map集合）
//        // 加载xml
//        InputStream resourceAsStream = BeanFactory.class.getClassLoader().getResourceAsStream("beans.xml");
//        // 解析xml
//        SAXReader saxReader = new SAXReader();
//        try {
//            Document document = saxReader.read(resourceAsStream);
//            Element rootElement = document.getRootElement();
//            List<Element> beanList = rootElement.selectNodes("//bean");
//            for (int i = 0; i < beanList.size(); i++) {
//                Element element =  beanList.get(i);
//                // 处理每个bean元素，获取到该元素的id 和 class 属性
//                String id = element.attributeValue("id");        // accountDao
//                String clazz = element.attributeValue("class");  // com.lagou.edu.dao.impl.JdbcAccountDaoImpl
//                // 通过反射技术实例化对象
//                Class<?> aClass = Class.forName(clazz);
//                Object o = aClass.newInstance();  // 实例化之后的对象
//
//                // 存储到map中待用
//                map.put(id,o);
//
//            }
//
//            // 实例化完成之后维护对象的依赖关系，检查哪些对象需要传值进入，根据它的配置，我们传入相应的值
//            // 有property子元素的bean就有传值需求
//            List<Element> propertyList = rootElement.selectNodes("//property");
//            // 解析property，获取父元素
//            for (int i = 0; i < propertyList.size(); i++) {
//                Element element =  propertyList.get(i);   //<property name="AccountDao" ref="accountDao"></property>
//                String name = element.attributeValue("name");
//                String ref = element.attributeValue("ref");
//
//                // 找到当前需要被处理依赖关系的bean
//                Element parent = element.getParent();
//
//                // 调用父元素对象的反射功能
//                String parentId = parent.attributeValue("id");
//                Object parentObject = map.get(parentId);
//                // 遍历父对象中的所有方法，找到"set" + name
//                Method[] methods = parentObject.getClass().getMethods();
//                for (int j = 0; j < methods.length; j++) {
//                    Method method = methods[j];
//                    if(method.getName().equalsIgnoreCase("set" + name)) {  // 该方法就是 setAccountDao(AccountDao accountDao)
//                        method.invoke(parentObject,map.get(ref));
//                    }
//                }
//
//                // 把处理之后的parentObject重新放到map中
//                map.put(parentId,parentObject);
//
//            }
//
//
//        } catch (DocumentException e) {
//            e.printStackTrace();
//        } catch (ClassNotFoundException e) {
//            e.printStackTrace();
//        } catch (IllegalAccessException e) {
//            e.printStackTrace();
//        } catch (InstantiationException e) {
//            e.printStackTrace();
//        } catch (InvocationTargetException e) {
//            e.printStackTrace();
//        }
//
//    }

    static {
        //扫描指定包下的所有类
        Set<Class<?>> classSet = ClassUtils.getClasses("com.lagou.edu", true);

        try {
            //查找所有使用@Service注解标签类，并实例化
            findServiceAnnotation(classSet);
        } catch (Exception e) {
            e.printStackTrace();
        }

        //遍历注入对象
        for (String key : map.keySet()) {
            try {
                attributeAssign(map.get(key));
            } catch (IllegalAccessException e) {
                e.printStackTrace();
            }
        }

        //查找所有使用@Transactional注解，并实例化
        try {
            findTransactionalAnnotation(classSet);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }


    /**
     * 查找@Service 注解类并实例化对象到map中
     */
    private static Map<String, Object> findServiceAnnotation(Set<Class<?>> classes) throws Exception {
        for (Class<?> clazz : classes) {
            Service annotation = clazz.getAnnotation(Service.class);
            if (annotation != null) {
                Object obj = clazz.newInstance();
                //key为接口类全限定类名称
                if (clazz.getInterfaces() != null && clazz.getInterfaces().length > 0) {
                    String facesName = clazz.getInterfaces()[0].getName();
                    map.put(facesName, obj);
                }
                //key为类名首字母小写
                if (StringUtils.isNullOrEmpty(annotation.value())) {
                    String beanId = ClassUtils.toLowerCaseFirstOne(clazz.getSimpleName());
                    map.put(beanId, obj);
                }
                //key为设定值
                else {
                    map.put(annotation.value(), obj);
                }
                continue;
            }
        }
        return map;
    }

    /**
     * 依赖注入，为使用@Autowired注解标签的属性注入bean对象
     */
    private static void attributeAssign(Object obj) throws IllegalAccessException {
        Field[] fields = obj.getClass().getDeclaredFields();
        for (Field field : fields) {
            Autowired autowired = field.getAnnotation(Autowired.class);
            if (autowired != null) {
                String fieldName = field.getName();
                //获取到依赖注入的bean
                Object bean = map.get(fieldName);
                if (bean == null) {
                    bean = map.get(field.getType().getName());
                }

                if (bean != null) {
                    field.setAccessible(true);
                    field.set(obj, bean);
                }
            }
        }
    }


    /**
     * 查找@Transactional 注解类并实例化对象到map中
     */
    private static Map<String, Object> findTransactionalAnnotation(Set<Class<?>> classes) throws Exception {

        map.entrySet().forEach(x -> {
            String key = x.getKey();
            Object obj = x.getValue();
            //判断类是否是事务
            Transactional transactional = obj.getClass().getAnnotation(Transactional.class);
            Method[] methods = obj.getClass().getDeclaredMethods();
            boolean isTranMethods = Arrays.stream(methods).anyMatch(y -> y.isAnnotationPresent(Transactional.class));

            if (transactional != null || isTranMethods) {
                ProxyFactory proxyFactory = (ProxyFactory) map.get(ClassUtils.toLowerCaseFirstOne(ProxyFactory.class.getSimpleName()));
                //实现接口的代理类，使用jdk方式
                if (obj.getClass().getInterfaces().length > 0) {
                    obj = proxyFactory.getJdkProxy(obj);
                }
                //非实现接口的代理类，使用cglib
                else {
                    obj = proxyFactory.getCglibProxy(obj);
                }
                map.put(key, obj);
            }

        });


        return map;
    }


    public static Object getBean(String id) {
        return map.get(id);
    }

}
